from langchain_community.document_compressors.dashscope_rerank import DashScopeRerank

sequence = ["量子计算是一种利用量子态进行信息处理的新型计算方式",
    "计算机科学是一门研究计算理论和算法的学科",
    "量子比特是量子计算的基本单位，可以处于叠加状态",
    "人工智能被广泛应用于图像识别领域",]

reranker = DashScopeRerank(
    model="gte-rerank",
    # api_key= "sk-d16b46d66abb45bb960bd9c57804e2f9",
dashscope_api_key="sk-d16b46d66abb45bb960bd9c57804e2f9",
    # other params...
)
print(reranker.rerank(documents=sequence, query="量子计算的基本原理", top_n=5))